// Copyright (c) 2025, 大连理工大学 (Dalian University of Technology)
//
// Licensed under the Mulan PSL v2.
// You can use this file according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of the License at
//
//     http://license.coscl.org.cn/MulanPSL2
//
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
// EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. SEE THE MULAN PSL v2
// FOR MORE DETAILS.

#include "Import/BDFImport/BulkData/Constraint/RBE3Parser.h"

#include "gtest/gtest.h"

#include "DataStructure/Common/Constant.h"

#include "DBManager/DBServiceFactory.h"
#include "DBManager/IService/Input/Constraint/IRBE3Service.h"

#include "Import/BDFImport/BDFParserBase.h"

using SG::DataStructure::Common::TOLERANCE;

using namespace SG::DBManager;
using namespace SG::Import;
using namespace SG::DataStructure;

extern std::shared_ptr<DBServiceFactory> DB_SERVICEFACTORY;
extern std::vector<std::string>          dividedEntryInfo;

class RBE3ParserTest : public ::testing::Test
{
  public:
    static void SetUpTestSuite ()
    {
        DB_SERVICEFACTORY->ClearDatabase ();
    }
};

TEST_F (RBE3ParserTest, ParseEntry1)
{
    std::vector<std::string> Entry = { "RBE3           1            1000  123456      1.  123456      25      20+0000051",
                                       "+0000051      15      10       5      46      41      36      31      26+0000052" };  //,
    // "+0000052   ALPHA   2.2-5" };
    BDF::divideBulkEntry (Entry, dividedEntryInfo);

    SG::Import::BDF::BULKDATA::ParseRBE3 (DB_SERVICEFACTORY, dividedEntryInfo);
    SG::DataStructure::FEM::RBE3Data data;

    DB_SERVICEFACTORY->get<IRBE3Service> ()->FindById (1, data);

    ASSERT_TRUE (data.m_id == 1);
    ASSERT_TRUE (data.m_refc == 123456);
    ASSERT_TRUE (data.m_refg = 1000);
    ASSERT_TRUE (data.m_WTCG.size () == 1);
    ASSERT_NEAR (data.m_WTCG[0].m_wt1, 1.0, TOLERANCE);
    ASSERT_TRUE (data.m_WTCG[0].m_g.size () == 10);
    ASSERT_TRUE (data.m_WTCG[0].m_c == 123456);
}

TEST_F (RBE3ParserTest, ParseEntry2)
{
    std::vector<std::string> Entry = { "RBE3           2            1000  123456      1.  123456      25      20+0000051",
                                       "+0000051      15      10       5      46      41      36      31      26+0000052",
                                       "+0000052   ALPHA   2.2-5" };
    BDF::divideBulkEntry (Entry, dividedEntryInfo);

    SG::Import::BDF::BULKDATA::ParseRBE3 (DB_SERVICEFACTORY, dividedEntryInfo);
    SG::DataStructure::FEM::RBE3Data data;

    DB_SERVICEFACTORY->get<IRBE3Service> ()->FindById (2, data);

    ASSERT_TRUE (data.m_id == 2);
    ASSERT_TRUE (data.m_refc == 123456);
    ASSERT_TRUE (data.m_refg = 1000);
    ASSERT_TRUE (data.m_WTCG.size () == 1);
    ASSERT_NEAR (data.m_WTCG[0].m_wt1, 1.0, TOLERANCE);
    ASSERT_TRUE (data.m_WTCG[0].m_g.size () == 10);
    ASSERT_TRUE (data.m_WTCG[0].m_c == 123456);
    ASSERT_NEAR (data.m_alpha, 2.2E-5, TOLERANCE);
}
